import pandas as pd
from typing import List


def calculate_total_traffic(file_paths: List[str]):
    res = []
    for file_path in file_paths:
        t = pd.read_csv(file_path, sep='|', dtype=str)
        t = t[t['用户账号'].isin(account['用户账号'])]
        res.append(t)
    return pd.concat(res)


from pathlib import Path

# 指定目录路径
directory = r"D:\家宽\Radius\temp"

# 创建Path对象
dir_path = Path(directory)

# 确保目录存在
if not dir_path.exists():
    print(f"错误：目录 {directory} 不存在")
else:
    # 递归查找所有CSV文件
    csv_files = list(dir_path.rglob('*.csv'))

    # 打印结果
    if csv_files:
        print(f"在目录 {directory} 及其子目录中找到 {len(csv_files)} 个CSV文件：")
        for i, file_path in enumerate(csv_files, 1):
            print(f"{i}. {file_path}")
    else:
        print(f"在目录 {directory} 及其子目录中未找到任何CSV文件")

account = pd.read_csv("/temp/t.csv", dtype=str)
df = calculate_total_traffic(csv_files)
df.to_csv("/temp/指定账号.csv")
